package com.bat.conf.datasource;

/**
 * @Description: 数据源Key管理
 * @Author: zpf
 * @Date: 2020/1/16 下午9:33
 */
public class DynamicDataSourceContextHolder {
    private static final ThreadLocal<DatabaseTypeEnum> contextHolder = new ThreadLocal<>();

    /**
     * 获取key
     *
     * @return 获取数据源key
     */
    public static DatabaseTypeEnum getDatabaseType() {
        return contextHolder.get();
    }

    public static void setDatabaseType(DatabaseTypeEnum type) {
        contextHolder.set(type);
    }

    /**
     * 重置默认数据源
     */
    public static void resetDatabaseType() {
        contextHolder.set(DatabaseTypeEnum.LOCAL);
    }

    /**
     * 清理绑定
     */
    public static void clear() {
        contextHolder.remove();
    }
}
